Codexshaper / wpb-framework

function menuFix(slug) {

    var $ = jQuery;

    let menuRoot = $('#toplevel_page_' + slug);

    let currentUrl = window.location.href;

    let currentPath = currentUrl.substr( currentUrl.indexOf('admin.php') );

    menuRoot.on('click', 'a', function() {

        var self = $(this);

        $('ul.wp-submenu li', menuRoot).removeClass('current');

        if ( self.hasClass('wp-has-submenu') ) {

            $('li.wp-first-item', menuRoot).addClass('current');

        } else {

            self.parents('li').addClass('current');

        }

    });

    $('ul.wp-submenu a', menuRoot).each(function(index, el) {

        if ( $(el).attr( 'href' ) === currentPath ) {

            $('ul.wp-submenu a', menuRoot).parent().removeClass('current');

            $(el).parent().addClass('current');

        }

    });

}

export default menuFix;